package net.hserver.admin.system.service;

import cn.hserver.core.ioc.annotation.Autowired;
import cn.hserver.core.ioc.annotation.Bean;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.mybatisflex.core.paginate.Page;
import com.mybatisflex.core.query.QueryWrapper;
import lombok.extern.slf4j.Slf4j;
import net.hserver.admin.system.common.MyCode;
import net.hserver.admin.system.entity.Role;
import net.hserver.admin.system.entity.RoleMenu;
import net.hserver.admin.system.entity.UserRole;
import net.hserver.admin.system.mapper.RoleMapper;
import net.hserver.admin.system.mapper.RoleMenuMapper;
import net.hserver.admin.system.mapper.UserRoleMapper;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Set;

/**
 * 角色信息 服务类
 *
*
 * @since 2022-06-12
 */
@Slf4j
@Bean
public class RoleService {

    @Autowired
    private RoleMapper mapper;
    @Autowired
    private RoleMenuMapper roleMenuMapper;
    @Autowired
    private UserRoleMapper userRoleMapper;

    private final String CACHE_ROLE_MENUS = "roleMenus";
    /**
     * 分页查询列表
     *
     * @param page   分页对象
     * @param entity 查询实体
     * @return 所有数据
     */
    public Page<Role> page(Page<Role> page, Role entity) {
        return mapper.paginate(page,new QueryWrapper().like(Role::getRoleName, entity.getRoleName(), StrUtil.isNotEmpty(entity.getRoleName())).orderBy(Role::getRoleId,false));
    }

    /**
     * 不分页查询列表
     *
     * @param entity 实体对象
     * @return 查询结果
     */
    public List<Role> list(Role entity) {
        return mapper.selectListByQuery(new QueryWrapper());
    }

    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    public Role getById(Long id) {
        return mapper.selectOneById(id);
    }

    /**
     * 新增单条数据
     *
     * @param entity 实体对象
     */
    public void save(Role entity) {
        checkRoleNameUnique(entity);
        mapper.insert(entity);
        insertRoleMenu(entity);
    }

    /**
     * 修改单条数据
     *
     * @param entity 实体对象
     */
    public void update(Role entity) {
        checkRoleNameUnique(entity);
        // 删除角色与菜单关联
        roleMenuMapper.deleteRoleMenuByRoleId(entity.getRoleId());
        mapper.update(entity);
        insertRoleMenu(entity);
    }

    /**
     * 批量删除数据
     *
     * @param ids 主键集合
     */
    public void deleteByIds(List<Long> ids) {
        long l = userRoleMapper.selectCountByQuery(QueryWrapper.create().in(UserRole::getRoleId, ids));
        if (l>0) {
            throw MyCode.DATA_VERIFY_ERROR.toException("角色已绑定用户，无法删除");
        }
        roleMenuMapper.deleteRoleMenu(ids);
        mapper.deleteBatchByIds(ids);
    }

    /**
     * 新增角色菜单信息
     *
     * @param role 角色对象
     */
    public void insertRoleMenu(Role role) {
        // 新增用户与角色管理
        if (CollUtil.isNotEmpty(role.getMenuIds())) {
            List<RoleMenu> list = new ArrayList<>();
            for (Long menuId : role.getMenuIds()) {
                RoleMenu rm = new RoleMenu();
                rm.setRoleId(role.getRoleId());
                rm.setMenuId(menuId);
                list.add(rm);
            }
            if (!list.isEmpty()) roleMenuMapper.batchRoleMenu(list);
        }
    }

    /**
     * 校验角色名称是否唯一
     *
     * @param role 角色信息
     */
    public void checkRoleNameUnique(Role role) {
        Long roleId = Objects.isNull(role.getRoleId()) ? Long.valueOf(-1L) : role.getRoleId();
        Role info = mapper.checkRoleNameUnique(role.getRoleName());
        if (Objects.nonNull(info) && info.getRoleId().longValue() != roleId.longValue()) {
            throw MyCode.DUPLICATE_KEY.toException("角色名称已存在");
        }
    }

    public Set<Long> selectMenuIdsByRoleId(Long roleId) {
        return roleMenuMapper.selectMenuIdsByRoleId(roleId);
    }

}
